calculate area of square in python